package com.yming.tools;

import dev.langchain4j.agent.tool.P;
import dev.langchain4j.agent.tool.Tool;
import dev.langchain4j.agent.tool.ToolMemoryId;
import org.springframework.stereotype.Component;

@Component
public class CalculatorTools {
    @Tool(name = "加法", value = "将参数a和b相加并返回运算结果")
    double sum(
            @ToolMemoryId String memoryId,
            @P(value = "加数1", required = true) double a,
            @P(value = "加数2", required = true) double b
    ) {
        System.out.println("调用加法运算，memoryId = " + memoryId);
        return a + b;
    }

    @Tool(name = "平方根", value = "求x的平方根")
    double squareRoot(
            @ToolMemoryId String memoryId,
            @P(value = "求数", required = true) double x
    ){
        System.out.println("调用平方根运算，memoryId = " + memoryId);
        return Math.sqrt(x);
    }
}
